legend function in python